import torch
import torch.nn as nn
from yolov1 import yolov1

batch = torch.zeros(10, 3, 448, 448)

model = yolov1()
out = model(batch)
print(out)
